package ipad.asia.utils;

import org.apache.commons.net.ftp.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.*;
@Component
public class FtpUtil {
    /**
     * Description: 向FTP服务器上传文件
     *
//     * @param host     FTP服务器hostname
//     * @param port     FTP服务器端口
//     * @param username FTP登录账号
//     * @param password FTP登录密码
//     * @param basePath FTP服务器基础目录
//     * @param filePath FTP服务器文件存放路径。文件的路径为basePath+filePath
//     * @param FileName 上传到FTP服务器上的文件名
    * salt 为用户表中的唯一随机盐，用户身份标志
     * @param input    输入流
     * @return 成功返回true，否则返回false
     *
     */
    public static boolean uploadFile( String filename, InputStream input) {

        String host = "172.81.238.137";
        int port = 21;
        String username = "feng";
        String password = "6369180fdw";
        String basePath = "/upload";
        String filePath = "/shop";
//        String suffix = FileName.substring(FileName.lastIndexOf(".")+1).trim();
//        String filename = MD5Util.nameToMD5(FileName,salt)+"."+suffix;
//        System.out.println(filename);
        boolean result = false;
        FTPClient ftp = new FTPClient();
        ftp.setAutodetectUTF8(true);
        try {
            int reply;
            ftp.connect(host, port);// 连接FTP服务器
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(username, password);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return result;
            }
            //切换到上传目录
            if (!ftp.changeWorkingDirectory(basePath + filePath)) {
                //如果目录不存在创建目录
                String[] dirs = filePath.split("/");
                String tempPath = basePath;
                for (String dir : dirs) {
                    if (null == dir || "".equals(dir)) continue;
                    tempPath += "/" + dir;
                    if (!ftp.changeWorkingDirectory(tempPath)) {  //进不去目录，说明该目录不存在
                        if (!ftp.makeDirectory(tempPath)) { //创建目录
                            //如果创建文件目录失败，则返回
                            System.out.println("创建文件目录" + tempPath + "失败");
                            return result;
                        } else {
                            //目录存在，则直接进入该目录
                            ftp.changeWorkingDirectory(tempPath);
                        }
                    }
                }
            }
            //设置上传文件的类型为二进制类型
            ftp.setFileType(FTP.BINARY_FILE_TYPE);
            //上传文件
            if (!ftp.storeFile(filename, input)) {
                return result;
            }
            input.close();
            ftp.logout();
            result = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }

    /**
     * Description: 从FTP服务器下载文件
     *
//     * @param host       FTP服务器hostname
//     * @param port       FTP服务器端口
//     * @param username   FTP登录账号
//     * @param password   FTP登录密码
//     * @param remotePath FTP服务器上的相对路径
//     * @param FileName   要下载的文件名,MD5加用户盐密确保唯一
     * salt 为用户表中的唯一随机盐，用户身份标志
//     * @param localPath  下载后保存到本地的路径
     * @return
     */
    public static boolean downloadFile(String fileName) throws IOException {
//        String suffix = FileName.substring(FileName.lastIndexOf(".")+1).trim();
//        String fileName = MD5Util.nameToMD5(FileName,salt)+"."+suffix;
//        System.out.println(fileName);
        String host = "172.81.238.137";
        int port = 21;
        String username = "feng";
        String password = "6369180fdw";
        String basePath = "upload";
        String remotePath = "upload/shop";
        File directory = new File("");
        String localPath = directory.getCanonicalPath()+"/e-shop-api/src/main/resources/upload";
        boolean result = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(host, port);
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(username, password);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return result;
            }
            ftp.changeWorkingDirectory(remotePath);// 转移到FTP服务器目录
            FTPFile[] fs = ftp.listFiles();
            for (FTPFile ff : fs) {
                System.out.println(ff.getName());
                if (ff.getName().equals(fileName)) {
                    File localFile = new File(localPath + "/" + ff.getName());
                    OutputStream is = new FileOutputStream(localFile);
                    ftp.retrieveFile(ff.getName(), is);
                    is.close();
                    System.out.println("下载的文件在服务器存在！");
                }
            }
            ftp.logout();
            result = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }
    //ftp上传文件测试main函数

}
